# 复制标签的文件txt到另一文件
import os
import shutil

def copy_matching_files(img_folder, txt_folder, destination_folder):
    # 列出第一个文件夹里的所有图片文件名（去除后缀）
    img_names = [os.path.splitext(img)[0] for img in os.listdir(img_folder) if img.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.gif'))]

    # 在第二个文件夹里找到与图片除后缀外名字相同的txt文件，并复制到第三个文件夹
    for img_name in img_names:
        txt_path = os.path.join(txt_folder, img_name + '.txt')
        if os.path.exists(txt_path):
            shutil.copy2(txt_path, destination_folder)
            print(f"Copied {txt_path} to {destination_folder}")

# 使用
img_folder = "F:\Dataset\sumdata(yes)\SumData\light-set\images\highlight-val"          # 替换为你的图片文件夹路径
txt_folder = "F:\Dataset\originData_train3\labels/val"            # 替换为你的txt文件夹路径
destination_folder = "F:\Dataset\sumdata(yes)\SumData\light-set\labels\highlight-val"   # 替换为你想要复制到的文件夹路径
copy_matching_files(img_folder, txt_folder, destination_folder)
